# -*- coding: utf-8 -*-
"""
@Time    : 2025/3/7 11:33 
@Author  : ZhangShenao 
@File    : 1.调用HuggingFace在线模型.py
@Desc    : 调用HuggingFace在线模型
"""
import os

import dotenv
import requests

# 加载环境变量,并读取HuggingFace的api_token
dotenv.load_dotenv()
api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")

# 定义API的base_url
# https://api-inference.huggingface.co/models 这个前缀是固定形式
# uer/gpt2-chinese-cluecorpussmall 这个是模型的namespace+名称
base_url = "https://api-inference.huggingface.co/models/uer/gpt2-chinese-cluecorpussmall"

# 在Http Header中指定api_token
headers = {"Authorization": f"Bearer {api_token}"}

# 发送文本生成请求
response = requests.post(base_url, headers=headers, json={"inputs": "帮我介绍下李白："})

# 打印结果
print(response.json())
